Update main.py

removed functions I no longer use and cleaned up the class