Lab2 corrected some code & added comments
Finished Lab2 code