Add updateUI method