very minor changes main.py