Перемешать элементы двумерного списка

Напишите программу, которая с помощью модуля random перемешивает случайным образом содержимое матрицы (двумерного списка).

Примечание. Выводить содержимое матрицы не нужно.

import random as rnd
matrix = [[1, 2, 3, 4],
          [5, 6, 7, 8],
          [9, 10, 11, 12],
          [13, 14, 15, 16]]

n, m, lst = len(matrix), len(matrix[0]), sum(matrix, [])
rnd.shuffle(lst)
matrix = [[lst[i * m + j] for j in range(m)] for i in range(n)]