Updated setup script to take a vagrant parameter.