working model relationships
removed unneeded import