Added category control