update yaml generator