Update main.py

Removed useless import