Add code coverage to solution