Changing way to import models for virtrm (II)