update main.py so it has no syntax error(actually i need to do it before last commit.)