Add saving to YAML