Make setup script smarter.