Rishabh Jain's submission of Square | CodeChef