Added assignment 3 problem 1