Small UI touch-ups