main.py file cleanup