Update main.py

Drastical name change in order to save the logic!