Let me know if you need help with the above, it's probably a bit daunting to figure out where to change `main.py` for this.