Added entity references to model classes