Add support for YAML input