adding in initial category handling