Fixed indent bugs in main.py

My bad. I indented a few blocks incorrectly.