changed the container class in jQuery code