add inject class support