Python toy program to understand the basics of ray tracing