Adding some integration tests