added ability to update UI