Added code to setup the availability zone