Update partitionwork.ipynb