Add association on Comment model