addded test class