Initial setup for part 2