Add coverage to tests