wip: add return_square for checks