Add some UI changes