US State Dept: We expect return to Iran nuclear deal 'within a short time'