Add integration testing

I added a test case with the example code.