Working on part 5 of Assignment 1