added some lines in test