predictions saving and loading works