Add UI changes