Handle exit() within END blocks better