updates based on new UI