US will need to work with China to make progress with North Korea says official