Replaced more instances of HasTraits.set with trait_set