UI changes to make things look nice