Add comment model and associations