added traits for model