Update generator to include transition mixin