Modified a few UI things