slight UI changes