jQuery beforeunload

Sveiki,

rašiau, rašiau ir pasirašiau nežinau ar labai jau vykusį kodą. Jo veikimo principas yra toks, kad pakeitus formoje kažkokius duomenis ir bandant išeiti iš puslapio (kuriame yra ta forma), jį perkrauti ar išjungti naršyklę - gaunamas pranešimas [neišsaugoti duomenys bus prarasti]. Viskas čia veikė jau nuo pat pradžių, tačiau vėliau pastebėjau, kad spaudžiant mygtuką submit taip pat gaunamas toks pranešimas, taigi čia ir prasidėjo kūrimas :D padariau kintamajį, kuris keičiasi užvedus kursorių ant submit mygtuko. Kodą pateikiu žemiau ir norėčiau sužinoti ar būtų galima jį kažkaip optimizuoti, ar čia viską gerai ir padariau?

           if ($('.form').length > 0) {
	var displayMessage = true;

	$('.form').data('serialize', $('.form').serialize());
	$(window).bind('beforeunload', function(e) {
	    if ($('.form').serialize() != $('.form').data('serialize') && displayMessage) {
	    	return unload_page_message;
	    } else {
	    	e = null;
	    }
	});

	$('form .save').hover(
       	function(){ displayMessage = false; },
       	function(){ displayMessage = true; } 
	);
    }

P.S. pastebėjau, kad ir WP kažką panašaus turi, tik nelabai radau, kaip ten padaryta...

var inFormOrLink;
$('a').live('click', function() { inFormOrLink = true; });
$('form').bind('submit', function() { inFormOrLink = true; });

$(window).bind("beforeunload", function() { 
   return inFormOrLink || confirm("Do you really want to close?"); 
})