Update main.py

Removed redundant code.