Erfan Alimohammadi's submission of Square | CodeChef