Small UI fixes.
Small UI fixes.