Add safe_attributes handling