Updated file with Part 2 code