Styled code to conform to pep8