corrected main.py file for the main creation.