Adding test cover