Unit testing exercise