Use Numpy save format instead of pickle