Wrote two integration tests