Add problem 3 and changes