Cleaned up code in and added comments to lab3