Adding use statement for MorphTo relationship