Added test utils class