Fixed formatting to better match PEP8