Add support for FrameworkExtension