IAEA chief says ’more work needed’ with Iran after visit