grab this cool change to main.py

Y U NO MERGE?