Conformed the code to pep8