assignment 6 - add problem 1 solution