assignment 5 - add problem 4 and 5