update for code coverage and tests