Added some code to allow squares to collide