add model pruner :-)