Python Matrix Coding Questions