Add re-training, separate model pruning, and bump version