Test class addded

Done:
1. Test class added

To do:
1. Try more