United States interested in working with Imran Khan's government