Change Javascript to jQuery