Update UI code per API changes