code added for UI
UI code added