Fixed code to conform with what pep8 is expecting.