Changed file to conform pep-8