Update main.py

broke the interface