Added assign_attributes for update