Update main.py

removed functions I no longer use and cleaned up the class
Merge pull request #2 from c4ts4thewin/function-cleanup

Update main.py