Added some code to prevent agents from moving out of board (I guess). Still, visualisation doesn't work :(