Completing our first unit test suite