Cleaned up code to comply with PEP-8, per PyCharm's inspection.