added tests :)