Update OnOffTrait.scala