update it to use JQuery