Reuse model from train on test, update maps