Simple python script for raytracing