Added correct version of main.py file