Add ability to set the availability zone of a node