Add support for documentWillSave