Cleaned up formatting (PEP8)