added more model relations