Fix serialization, changed from Pickle to jsonpickle