Categorization bit further, save not working yet