matrix : hw5.py all pass