Update to use vagrant box