Add in better ways to pickle tasks