fix code to comply with pep8