Add unit testcases