Added ability to use custom generator templates