Updated visualization for single lens.

Changes:
- Updated to use new mainWFS.py which has the lenslet array for the numerical solution
- Improved plot aesthetics
- Removed SingleLens.py which is no longer needed
- Attempted to improve readability and modularity
I think you can remove this as it is not needed anymore.