Added comment model with associations