Added floor and collision detection