Add coverage code