Add unit test
Create test unit