Update main.py

Unimportant stuff added.