The US president says Tehran has to do it "otherwise we just go back and do what we have to do"