Unit test added