Clean up code, PEP8.