updated to follow pep8