Remove Traits class