Shortened main.py

Basically figured out who to take a bunch of lines of code out.