Update main.py

for testing only