Assignment 3 - Problem 4 - started