Added category model and association