Clean up notebooks (error_correction_model)