Added tensor components in writing/loading the model