making code coverage over 90%