Better output when category column removed