Tests added for converage