Small UI style fixes