Added Extend, Implements and Traits to test class