adding traits handling