added unit test update. coverage still at 97%