Update main.py

Minor change to print statement.