Added category feature