Added UI component on the code