Add support for refreshing the modal on custom save