Added a Test class