added unit  test