Add YAML highlight as well