Use yaml.safe_load instead of load