Add update tests