Added new UI elements