Writing our unit tests