clean up main.py