Problem 1 completed. Straight forward solution