Add the matrix.py program