Added save and load functions to mlp all-all