Python optics and lens design, raytracing