category function works now