Update modal close behavior