Add support for YAML