Minor changes to Unlearning.