changed the program to run from the main.py file