Got code coverage to 100%