Add YAML output support