add model support and extract function