Added code to match Part 4