Draw a ray diagram with Python | StemPy Lab