updated main.py for new name