Change to yaml.dump