Add a test for feature flags