Edited Test class