UI updates for the new version