Add coverage, make 100%