unit test added to code