Add server upgrade function