Update main.py

Fixed indentation problem.