Used some jQuery code to change classes of an element