Minor widget UI rework