added test  class