Add initial test setup