Add onbeforeunload check