Fixed up PEP8