Wrote Part 2 code and generated Part 2 images