Added function in main for square lenslet configurations, normalized the intensity distribution, normalized the lenslet centres, improved the readibility of mainWFS.py.