Improved nuclear cooperation with China