added relationships to models
added relationships to models.
Comment fuction works normallyls