made changes for part 2