[IM] Small tweaks and fixes on UI