made some minor changes related to UI