Modify yaml output