added association for category model