Add basic setup for tests