Change models files from .pkl to .xr