Small changes to update