Add ability to override the generator