Editing: AddTrait.php