After criticism, US president says he's 'okay' with following international law, but warns Tehran will 'suffer consequences' if it acts out of line