Added trait tests