coding the matrix : fix hw0 problem3 miss understood the problem