Minor changes for ui