Add test setup