Did a small UI tweak