Add template files to generator