US operations in Iran will be completed in four to six weeks, White House press secretary Karoline Leavitt says.