Finish main work on 'lab2.py'

Main work on this file is finished and testing can begin