Update main.py

Undid attempt to add changes to fix #59