Maybe you could do it in the save() method.