Added test setup