Unit  tests added