added relationships to models