Adding the Unit Test