China to work with Pakistan for upgraded CPEC version