Add coverage to trais setup