Add category method to models. Use category instead of class on directory view.