Changes in Test class