added some UI changes