Added test for traits