Add result for part two
update source code