Update the setup script