Added some integration testing.