Add basic integration test