added more unit tests

got code coverage back over 90%