Cleaned up main.py even more.