Added minor changes in test.