Added UI/UX