Add FOR UPDATE support