Adding option to configure custom generator