Update main.py

Diggy