Update the jquery code