Add edit and update feature