added check to comment model