Update problem 2,3,4 of project 2