cleaned code with pep8