main.py almost PEP8 compliant, fixed some 'wrong' spacing (three spaces), all tests passed