[ui] Layout tweaks from small screen