Adding more test to the unit test