Added unit test

added unit test in comment