Matrix Diagonal Sum — PY OPPE Question