Closes #2 homework