Fixed formatting to conform to PEP8