* add new argument for easy load prune program

* add argument make load pruned program more easier

* change save function arg

* fix test

* fix format