Add for_update method