updated jQuery code