Fixed name bug in main.py