Add JS for flash messages