Add upgrade flag to setup script