Add unit test part1