Updated code to follow pep8.