Iraq's PM returned to Baghdad after a two-day visit to Iran