add print_matrix.py