More changes to main.py