Set up Part 2 code