US wants to work with Shehbaz govt