Added basic unit test setup