Adding new UI changes