Updated diagram, still struggling with mocking KazooClient