Update jQuery code