update for the new UI