Finish Assignment 5 part 6