mod to pickle.load and that part of code.