Removed unnecessary code from main.py file