fix: moving save from numpy to pickle