add unit test part2