Add upgrade_gitlab function