Setting up unit test