Adding comments to code
Made pep8 changes