Adding class declaration case to CodeInjector, using a flag to do it