Added unit test setup