Added model linkage code