Add coverage tweaks

Coverage!