Modification of the Test class and adding method in the class Diaballik