Update main.py

Removed superfluous test code