Very minor change to main.py