Update Tim's main.py

Fixed the collide