update for new ui