Assignment 2 - Problem 2 completed