Added some better UI styling