Collision with walls now added