Added a basic collision with the walls