WIP Homework 5, missing problem 4