Add and test update_zone method