Create matrix_puzzle.py