Add support for generating YAML