Update main.py

Cleaned script