Added missing files for trait tests