Added  integration tests