Fixed some minor ui changes