Corrected some small errors in main.py