Update main.py

had to fix many bugs, know I'm gonna shorten my seminar paper for tomorrow