cleaned up code according to pep8 standards