diff --git a/README.md b/README.md
index 2249ae2..2c73af5 100644
--- a/README.md
+++ b/README.md
@@ -7,6 +7,14 @@ Tobias is your friendly PostgreSQL DBA who is obsessed with optimizing your data
 
 Named after a nerdy but friendly DBA who is obsessed with query performance, Tobias helps you find the optimal `work_mem` setting for your PostgreSQL queries. It runs your queries with various memory settings to determine the minimum `work_mem` needed to keep your queries entirely in memory without creating temporary files.
 
+## Prerequisites
+
+You'll need the huggingface CLI. On macOS you can install it via Homebrew:
+
+```shell
+$ brew install huggingface-cli
+```
+
 ## Installation
 
 ```shell
diff --git a/lib/tobias.rb b/lib/tobias.rb
index 49a0f1f..07d8bb7 100644
--- a/lib/tobias.rb
+++ b/lib/tobias.rb
@@ -11,6 +11,9 @@
   end
 end
 
+# See: https://github.com/ged/ruby-pg/issues/538#issuecomment-1591629049
+ENV["PGGSSENCMODE"] = "disable"
+
 require "bundler/setup"
 Bundler.require(:default)
 
@@ -21,6 +24,7 @@
 require "enumerable-stats"
 require "benchmark"
 require "parquet"
+require "parallel"
 require "tty-markdown"
 require "tty-table"
 
diff --git a/lib/tobias/cli.rb b/lib/tobias/cli.rb
index e7a5894..376c654 100644
--- a/lib/tobias/cli.rb
+++ b/lib/tobias/cli.rb
@@ -55,7 +55,7 @@ def profile(script)
         raise "Script not found at: #{script}"
       end
 
-      container = Container.new(code)
+      container = Container.new(code, database)
       results = {}
 
       parsed = TTY::Markdown.parse(<<~MARKDOWN)
diff --git a/lib/tobias/container.rb b/lib/tobias/container.rb
index 163e78d..15d0b49 100644
--- a/lib/tobias/container.rb
+++ b/lib/tobias/container.rb
@@ -1,9 +1,12 @@
 # frozen_string_literal: true
 
+require "open3"
+
 module Tobias
   class Container
-    def initialize(code)
+    def initialize(code, database)
       @code = code
+      @database = database
       @queries = Concurrent::Hash.new
       @sql = Concurrent::Hash.new
       @options = Concurrent::Hash.new
@@ -15,54 +18,46 @@ def initialize(code)
       eval(code, binding, __FILE__, __LINE__)
     end
 
-    def run_setup(context)
-      run_action(@setup, context)
-    end
+    module DefaultHelpers
+      def db
+        @database
+      end
 
-    def run_load_data(context)
-      Etc.nprocessors.times do
-        fork do
-          context.disconnect
-          run_action(@load_data, context)
+      def run_parallel(list = Etc.nprocessors.times, &block)
+        db.disconnect
+
+        Parallel.each(list, in_processes: Etc.nprocessors) do |item|
+          instance_exec(item, &block)
         end
       end
+    end
 
-      context.disconnect
-      Process.waitall
+    def run_setup
+      @database.run("CREATE EXTENSION IF NOT EXISTS pg_stat_statements")
+      run_action(@setup)
     end
 
-    def run_query(query, context)
-      sql = if query.is_a?(String)
-               query
-             else
-               run_action(query, context).sql
-             end
+    def run_query(query)
+      @database.run(run_action(query).sql)
+    end
 
-      context.run(sql)
+    def run_teardown
+      run_action(@teardown)
     end
 
-    def run_teardown(context)
-      run_action(@teardown, context)
+    def options
+      Struct.new(*@options.keys).new(*@options.values)
     end
 
-    def run_action(action, context)
-      options = Struct.new(*@options.keys).new(*@options.values)
+    def run_action(action)
       helpers = @helpers
 
-      context.class_eval do
+      class_eval do
+        include DefaultHelpers
         include helpers
-
-        def options=(new_options)
-          @options = new_options
-        end
-
-        def options
-          @options
-        end
       end
 
-      context.options = options
-      context.instance_eval(&action)
+      instance_eval(&action)
     end
 
     def queries
@@ -90,7 +85,11 @@ def load_data(&block)
     end
 
     def query(name, sql = nil, &block)
-      @queries[name] = sql || block
+      if sql.is_a?(String)
+        @queries[name] = Proc.new { sql }
+      else
+        @queries[name] = block || Proc.new { raise "No SQL provided for query '#{name}'" }
+      end
     end
   end
 end
diff --git a/lib/tobias/evaluations/base.rb b/lib/tobias/evaluations/base.rb
index e180288..9a1c807 100644
--- a/lib/tobias/evaluations/base.rb
+++ b/lib/tobias/evaluations/base.rb
@@ -32,8 +32,7 @@ def initialize(database, container, options)
       def run(&block)
         results = Concurrent::Array.new
 
-        container.run_setup(database)
-        container.run_load_data(database)
+        container.run_setup
         container.queries.each do |name, query|
           result = run_each(name, query)
           results << result if result
@@ -41,7 +40,7 @@ def run(&block)
 
         to_markdown(results)
       ensure
-        container.run_teardown(database)
+        container.run_teardown
       end
 
       def run_each(query)
diff --git a/lib/tobias/evaluations/work_mem.rb b/lib/tobias/evaluations/work_mem.rb
index cfc37a1..6bbd5b8 100644
--- a/lib/tobias/evaluations/work_mem.rb
+++ b/lib/tobias/evaluations/work_mem.rb
@@ -12,7 +12,7 @@ def current_work_mem
       end
 
       def description
-        "Optional work_mem settings"
+        "Optimal work_mem settings"
       end
 
       def run_each(name, query)
@@ -22,7 +22,7 @@ def run_each(name, query)
           database.transaction do
             database.run("SET LOCAL work_mem = '#{value.to_sql}'")
             database.select(Sequel.function(:pg_stat_reset)).first
-            container.run_query(query, database)
+            container.run_query(query)
 
             stats = database[:pg_stat_database].
               where(datname: Sequel.function(:current_database)).
diff --git a/scripts/stress.rb b/scripts/stress.rb
index b2ba048..d3ebf4b 100644
--- a/scripts/stress.rb
+++ b/scripts/stress.rb
@@ -1,25 +1,22 @@
 # frozen_string_literal: true
 
-option(:total_rows, 10_000_000)
+option(:total_rows, 1_000_000)
 
 setup do
-  create_table? :workmem_stress do
+  db.create_table? :workmem_stress do
     primary_key :id
     column :name, String
     column :value, Integer
     column :payload, String
+    column :created_at, :timestamp, default: Sequel::CURRENT_TIMESTAMP
   end
-end
-
-load_data do
-  loop do
-    break if from(:workmem_stress).count >= options.total_rows
 
-    100.times do
-      from(:workmem_stress).multi_insert(1000.times.map do
+  run_parallel do
+    (options.total_rows / Etc.nprocessors / 1000).times do
+      db.from(:workmem_stress).multi_insert(1000.times.map do
         {
-          name: "name_#{Random.rand(1..1000)}",
-          value: Random.rand(1..10_000),
+          name: "name_#{Random.rand(1..1_000_000)}",
+          value: Random.rand(1..1_000_000),
           payload: SecureRandom.hex(128)
         }
       end)
@@ -28,26 +25,37 @@
 end
 
 teardown do
-  drop_table(:workmem_stress)
+  db.drop_table(:workmem_stress)
 end
 
 query(:large_sort) do
-  from(:workmem_stress)
-    .select(:id, :name, :value, :payload)
-    .order(Sequel.desc(:payload))
-    .limit(10_000)
+  db.
+    from(:workmem_stress).
+    select(:id, :name, :value, :payload).
+    order(Sequel.desc(:payload)).
+    limit(10_000)
+end
+
+query(:large_sort_created_at) do
+  db.
+    from(:workmem_stress).
+    select(:id, :name, :value, :payload).
+    order(Sequel.desc(:created_at)).
+    limit(10_000)
 end
 
 query(:hash_aggregation) do
-  from(:workmem_stress)
-    .select(:name, Sequel.function(:avg, :value))
-    .select { count("*") }
-    .group(:name)
-    .limit(10_000)
+  db.
+    from(:workmem_stress).
+    select(:name, Sequel.function(:avg, :value)).
+    select { count("*") }.
+    group(:name).
+    limit(10_000)
 end
 
 query(:self_join) do
-  from(Sequel.as(:workmem_stress, :a)).
+  db.
+    from(Sequel.as(:workmem_stress, :a)).
     join(Sequel.as(:workmem_stress, :b), id: :id).
     where { Sequel[:a][:id] < 1000 }.
     where { Sequel[:b][:id] < 1000 }.
diff --git a/scripts/tpcc.rb b/scripts/tpcc.rb
index 3e65710..810300a 100644
--- a/scripts/tpcc.rb
+++ b/scripts/tpcc.rb
@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 query(:stock_by_warehouse_and_district) do
-  warehouse_id = from(:warehouse).
+  warehouse_id = db.from(:warehouse).
     order(Sequel.lit("RANDOM()")).
     limit(1).
     first[:w_id]
@@ -12,7 +12,7 @@
     limit(1).
     first[:d_id]
 
-  from(:stock).
+  db.from(:stock).
     join(:order_line, ol_w_id: :s_w_id, ol_i_id: :s_i_id).
     where(ol_w_id: warehouse_id, ol_d_id: district_id).
     where(Sequel.lit("s_quantity < ?", rand(100..500))).
@@ -23,7 +23,8 @@
 end
 
 query(:most_active_districts) do
-  from(:district).
+  db.
+    from(:district).
     join(:order_line, [[:ol_d_id, :d_id], [:ol_w_id, :d_w_id]]).
     group(:d_w_id, :d_id, :d_name).
     select(
diff --git a/scripts/vector.rb b/scripts/vector.rb
index bebdedb..1bbcb80 100644
--- a/scripts/vector.rb
+++ b/scripts/vector.rb
@@ -1,43 +1,70 @@
 # frozen_string_literal: true
 
-option(:total_vectors, 5_000_000)
-option(:vector_dimension, 1_536)
+require "fileutils"
+require "open3"
+require "tmpdir"
 
 helpers do
-  def random_vector(size: options.vector_dimension)
+  def random_vector(size: 1_536)
     Array.new(size) { rand(-1.0..1.0) }
   end
+
+  def download_from_hugging_face(repo)
+    stdout, status = Open3.capture2(
+      "which", "hf"
+    )
+
+    unless status.success?
+      raise "Failed to find huggingface CLI: #{stdout}"
+    end
+
+    stdout, status = Open3.capture2(
+      "hf", "download", repo, "--repo-type=dataset"
+    )
+
+    unless status.success?
+      raise "Failed to download #{repo}: #{stdout}"
+    end
+
+    # Return the local directory where the dataset is cached.
+    "#{ENV["HOME"]}/.cache/huggingface/hub/datasets--#{repo.gsub("/", "--")}"
+  end
 end
 
 setup do
-  run("CREATE EXTENSION IF NOT EXISTS vector")
+  db.run("CREATE EXTENSION IF NOT EXISTS vector")
 
-  dimensions = options.vector_dimension
-  create_table? :items do
+  db.create_table? :items do
     primary_key :id
-    column :embedding, "vector(#{dimensions})"
+    column :title, :text
+    column :text, :text
+    column :embedding, "vector(#{1_536})"
+    column :created_at, :timestamp, default: Sequel::CURRENT_TIMESTAMP
   end
-end
 
-load_data do
-  loop do
-    break if from(:items).count >= options.total_vectors
-
-    from(:items).multi_insert(1_000.times.map do
-      {
-        embedding: ::Pgvector.encode(random_vector)
-      }
-    end)
+  local_dir = download_from_hugging_face("KShivendu/dbpedia-entities-openai-1M")
+  run_parallel(Dir.glob("#{local_dir}/snapshots/**/data/*.parquet")) do |file|
+    Parquet.each_row(file, columns: ["title", "text", "openai"]) do |row|
+      db.from(:items).insert(
+        title: row["title"],
+        text: row["text"],
+        embedding: Pgvector.encode(row["openai"])
+      )
+    end
   end
+
+  db.run("SET maintenance_work_mem = '128MB';")
+  db.run("CREATE INDEX IF NOT EXISTS items_embedding_idx ON items USING ivfflat (embedding) WITH (lists = 100)")
 end
 
 teardown do
-  drop_table(:items)
-  run("DROP EXTENSION IF EXISTS vector")
+  db.drop_table(:items)
+  db.run("DROP EXTENSION IF EXISTS vector")
 end
 
 query(:euclidean_nearest_neighbors) do
-  from(:items).
+  db.
+    from(:items).
     nearest_neighbors(
       :embedding,
       random_vector,
@@ -47,7 +74,8 @@ def random_vector(size: options.vector_dimension)
 end
 
 query(:cosine_nearest_neighbors) do
-  from(:items).
+  db.
+    from(:items).
     nearest_neighbors(
       :embedding,
       random_vector,
@@ -57,7 +85,8 @@ def random_vector(size: options.vector_dimension)
 end
 
 query(:inner_product_nearest_neighbors) do
-  from(:items).
+  db.
+    from(:items).
     nearest_neighbors(
       :embedding,
       random_vector,
diff --git a/spec/container_spec.rb b/spec/container_spec.rb
index 547908c..fdc71d1 100644
--- a/spec/container_spec.rb
+++ b/spec/container_spec.rb
@@ -3,16 +3,18 @@
 require "spec_helper"
 
 RSpec.describe Tobias::Container do
+  let(:database) { instance_double("Database") }
+
   describe "#initialize" do
     it "stores the provided code" do
       code = "query(:test) { puts 'hello' }"
-      container = described_class.new(code)
+      container = described_class.new(code, database)
 
       expect(container.instance_variable_get(:@code)).to eq(code)
     end
 
     it "initializes empty queries hash" do
-      container = described_class.new("")
+      container = described_class.new("", database)
 
       expect(container.queries).to eq({})
     end
@@ -24,25 +26,25 @@
         end
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
 
       expect(container.queries).to have_key(:simple_query)
       expect(container.queries[:simple_query]).to be_a(Proc)
     end
 
     it "handles empty code without error" do
-      expect { described_class.new("") }.not_to raise_error
+      expect { described_class.new("", database) }.not_to raise_error
     end
 
     it "raises error for invalid Ruby syntax" do
       invalid_code = "query(:broken { syntax error"
 
-      expect { described_class.new(invalid_code) }.to raise_error(SyntaxError)
+      expect { described_class.new(invalid_code, database) }.to raise_error(SyntaxError)
     end
   end
 
   describe "#query" do
-    let(:container) { described_class.new("") }
+    let(:container) { described_class.new("", database) }
 
     it "stores a query with the given name and block" do
       test_block = proc { "test query" }
@@ -77,7 +79,7 @@
 
   describe "#queries" do
     it "returns an empty hash when no queries are defined" do
-      container = described_class.new("")
+      container = described_class.new("", database)
 
       expect(container.queries).to eq({})
     end
@@ -88,7 +90,7 @@
         query(:orders) { from(:orders) }
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
 
       expect(container.queries.keys).to contain_exactly(:users, :orders)
     end
@@ -98,7 +100,7 @@
         query(:test) { "executable" }
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
 
       expect(container.queries[:test].call).to eq("executable")
     end
@@ -122,7 +124,7 @@
         end
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
       query_block = container.queries[:complex_query]
 
       expect(query_block).to be_a(Proc)
@@ -141,7 +143,7 @@
         end
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
       query_block = container.queries[:with_variables]
 
       expect(query_block).to be_a(Proc)
@@ -156,7 +158,7 @@
         end
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
       query_block = container.queries[:error_query]
 
       expect { query_block.call }.to raise_error(StandardError, "Query execution failed")
@@ -173,7 +175,7 @@ def helper_method
         end
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
 
       expect(container.queries[:with_helper].call).to eq("helper query")
     end
@@ -189,7 +191,7 @@ def helper_method
         end
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
 
       expect(container.queries[:isolated].call).to eq("isolated")
     end
@@ -222,7 +224,7 @@ def helper_method
         end
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
 
       expect(container.queries.keys).to contain_exactly(
         :stock_by_warehouse_and_district,
@@ -241,7 +243,7 @@ def helper_method
         query(:third) { "3" }
       RUBY
 
-      container = described_class.new(code)
+      container = described_class.new(code, database)
 
       # Ruby hash preserves insertion order in Ruby 2.7+
       expect(container.queries.keys).to eq([:first, :second, :third])
diff --git a/tobias.gemspec b/tobias.gemspec
index addcb24..8863204 100644
--- a/tobias.gemspec
+++ b/tobias.gemspec
@@ -33,6 +33,7 @@ Gem::Specification.new do |s|
   s.add_dependency "sequel", "~> 5.76", ">= 5.76.0"
   s.add_dependency "parquet", "~> 0.7", ">= 0.7.3"
   s.add_dependency "pgvector", "~> 0.3", ">= 0.3.0"
+  s.add_dependency "parallel", "~> 1.20", ">= 1.20.0"
   s.add_dependency "thor", "~> 1.3", ">= 1.3.0"
   s.add_dependency "tty-markdown", "~> 0.7", ">= 0.7.0"
   s.add_dependency "tty-table", "~> 0.12", ">= 0.12.0"