This is part 2 of the assignment