cleaned code and Pep8
Pep8 compliant