Add unit coverage