Add AddGenerator tests