Add collision detection for the floor