Added relations to comment model