add some integration testing