Added coverage to tests