Update main.py

I make a break, so I gonna save this.