minor edit to main.py