Made some visual tweaks to the UI