fixed ray trace for realistic lens and added visualizations
neaten it up a bit