clean up main.py

clean up