Switch note and category models?