simple raytracer in python