made slight UI changes