Changed UI design based on feedback