Update to use cloud-init