Make sure to click kinda to the bottom left of the back arrow.
This is the code for a colab with one of my friends! I gonna polish this now in like a week!