Allow update_attributes to accept a block