Add missing methods from generator