Make it more PEP8 compliant