Changed UI

Made it better