enforced pep8 and added comments
removed debugging print statement