Edited main.py to follow PEP8