Added UI changes
Changes