Move square methods to square controller