adding Units Tests