Style fixes for new UI