First set of unit test are working