Adding a few visual tweaks