Remove extra code that happened to be in main.py