Worked on ProblemSet4