Changes done to Test

changed main method