fixed error in part2