Fixed issue with not moving to squares - I had mixed up my rows and columns!