Added some code in Test.java