I think adding an outputVariable method to the generator is slightly neater