Adding unit test coverage and updating per github review