Ray tracer in Python