More assertions
New get_column function
Update get_square function
Update check sudoku and square to work with the new function