Fixing Python Styling (PEP8)