Major changes to main.py.