cleaning main.py