Updated the update feature