Make setup.sh a bit smarter