HW3: fix model issues for problem 3.