[IMP] remove pickle from code