Added some integration test