improvements on the new UI