Add onbeforeunload handler