Added required code for UI