Added UI restyling