Add update call and tests for update