Update main.py

Changed style of comments