Added cleanup code for the implemented work units.