Cleaning up repo, adding comments to code getting ready for submission