Update HW9 problem 1

fixing matrix inputs