Make this Lo Shu Magic into a 4 x 4 matrix instead of a 3 x 3 matrix? 

 

def is_lo_shu_magic_square(square)make this Lo Shu Magic into a 4 x 4 matrix instead of a 3 x 3 matrix? Every change I made cased the output to be falsedef is_lo_shu_magic_square(square):    #Check if the square is a valid 3x3 grid    if len(square) != 3 or any(len(row) != 3 for row in square):        return False    #Check if the square contains the numbers 1 through 9 excatly    nums = set(num for row in square for num in row)    if nums != set(range(1, 10)):        return False    #Calculate the sum of the diagonal, first row, and first column    diag_sum = square[0][0] + square[1][1] + square[2][2]    row_sum = sum(square[0])    col_sum = sum(row[0] for row in square)    #Check if all rows, columns, and the diagonal have the same sum    return all(sum(row) == row_sum and sum(row[i] for row in square) == col_sum               for i, row in enumerate(square)) and diag_sum == row_sumsquare = [[8, 1, 6], [3, 5, 7], [4, 9, 2]]is_magic_square = is_lo_shu_magic_square(square)print('Is the square a Lo Shu Magic Square? ', is_magic_square) # Output: Truesquare2 = [[8, 5, 6], [3, 1, 7], [4, 9, 2]]is_magic_square = is_lo_shu_magic_square(square2)