- removed PEP-8 errors
- removed unused variables in main