Added coverage to tests.