Unittest added

Unittest added