assignment 4 - add problem 2