fix part2
update code