Setting up some unit tests