Simple Raytracer in Python