Added test units

Added test units