Add YAML generator