improve category feature