Adding a bit more logic into the setup.sh script