diff --git a/.ameba.yml b/.ameba.yml
new file mode 100644
index 0000000..48d03ce
--- /dev/null
+++ b/.ameba.yml
@@ -0,0 +1,23 @@
+# This configuration file was generated by `ameba --gen-config`
+# on 2024-02-24 10:06:31 UTC using Ameba version 1.6.1.
+# The point is for the user to remove these configuration records
+# one by one as the reported problems are removed from the code base.
+
+# Problems found: 1
+# Run `ameba --only Naming/AccessorMethodName` for details
+Naming/AccessorMethodName:
+  Description: Makes sure that accessor methods are named properly
+  Excluded:
+    - src/lucky_flow/webless/element.cr
+  Enabled: true
+  Severity: Convention
+
+# Problems found: 5
+# Run `ameba --only Lint/UselessAssign` for details
+Lint/UselessAssign:
+  Description: Disallows useless variable assignments
+  ExcludeTypeDeclarations: false
+  Excluded:
+    - src/lucky_flow.cr
+  Enabled: true
+  Severity: Warning
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 5a030ab..d6d5471 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -2,7 +2,7 @@ name: Lucky Flow CI
 
 on:
   push:
-    branches: [ main ]
+    branches: [main]
   pull_request:
     branches: "*"
 
@@ -58,7 +58,7 @@ jobs:
       - uses: actions/checkout@v3
       - uses: browser-actions/setup-chrome@v1
         with:
-          chrome-version: stable
+          chrome-version: beta
         if: matrix.os == 'windows-latest'
       - uses: crystal-lang/install-crystal@v1
         with:
diff --git a/Dockerfile b/Dockerfile
index 2ad1665..5c0c15f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,9 +1,13 @@
-FROM crystallang/crystal:1.4.1
+FROM crystallang/crystal:1.5.1
 WORKDIR /data
 EXPOSE 3002
 
 RUN apt-get update \
-  && apt-get install -y libnss3 libgconf-2-4 chromium-browser \
+  && apt-get install -y \
+    libnss3 \
+    libgconf-2-4 \
+    chromium-browser \
+    firefox-geckodriver \
   && apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
 
 COPY . /data
diff --git a/script/setup b/script/setup
index 07cca0a..bec0049 100755
--- a/script/setup
+++ b/script/setup
@@ -4,12 +4,15 @@
 set -e
 set -o pipefail
 
-if ! command -v docker-compose > /dev/null; then
-  printf 'Docker and docker-compose are not installed.\n'
+if command -v docker-compose > /dev/null; then
+  docker-compose build
+  docker-compose run app shards install
+elif command -v docker compose > /dev/null; then
+  docker compose build
+  docker compose run app shards install
+else
+  printf 'Docker and/or docker-compose are not installed.\n'
   printf 'See https://docs.docker.com/compose/install/ for install instructions.\n'
   exit 1
 fi
 
-docker-compose build
-
-docker-compose run app shards install
diff --git a/script/test b/script/test
index 9475b61..f6ddfce 100755
--- a/script/test
+++ b/script/test
@@ -4,7 +4,11 @@
 set -e
 set -o pipefail
 
-COMPOSE="docker-compose run app"
+if command -v docker-compose > /dev/null; then
+  COMPOSE="docker-compose run app"
+elif command -v docker compose > /dev/null; then
+  COMPOSE="docker compose run app"
+fi
 
 printf "\nrunning specs with 'crystal spec'\n\n"
 $COMPOSE crystal spec
diff --git a/shard.yml b/shard.yml
index 214a090..7e7d247 100644
--- a/shard.yml
+++ b/shard.yml
@@ -4,7 +4,7 @@ version: 0.9.2
 authors:
   - Paul Smith <paulcsmith0218@gmail.com>
 
-crystal: "~> 1.4"
+crystal: "~> 1.5"
 
 license: MIT
 
diff --git a/spec/lucky_flow_spec.cr b/spec/lucky_flow_spec.cr
index 4e92afc..9d6866e 100644
--- a/spec/lucky_flow_spec.cr
+++ b/spec/lucky_flow_spec.cr
@@ -237,60 +237,120 @@ describe LuckyFlow do
     flow.should have_element("h1", text: "Target")
   end
 
-  it "can open screenshots", tags: "headless_chrome" do
-    flow = LuckyFlow.new
-    fake_process = FakeProcess
-    time = Time.utc
+  context "with headless chrome" do
+    it "can open screenshots", tags: "headless_chrome" do
+      flow = LuckyFlow.new
+      fake_process = FakeProcess
+      time = Time.utc
 
-    flow.open_screenshot(fake_process, time)
+      flow.open_screenshot(fake_process, time)
 
-    fake_process.shell.should be_true
-    fake_process.command.should eq "open ./tmp/screenshots/#{time.to_unix}.png"
-  end
+      fake_process.shell.should be_true
+      fake_process.command.should eq "open ./tmp/screenshots/#{time.to_unix}.png"
+    end
 
-  it "can open fullsize screenshots", tags: "headless_chrome" do
-    flow = LuckyFlow.new
-    fake_process = FakeProcess
-    time = Time.utc
+    it "can open fullsize screenshots", tags: "headless_chrome" do
+      flow = LuckyFlow.new
+      fake_process = FakeProcess
+      time = Time.utc
 
-    flow.open_screenshot(fake_process, time, fullsize: true)
+      flow.open_screenshot(fake_process, time, fullsize: true)
 
-    fake_process.shell.should be_true
-    fake_process.command.should eq "open ./tmp/screenshots/#{time.to_unix}.png"
-  end
+      fake_process.shell.should be_true
+      fake_process.command.should eq "open ./tmp/screenshots/#{time.to_unix}.png"
+    end
 
-  it "can reset the session", tags: "headless_chrome" do
-    flow = visit_page_with <<-HTML
-      <h1>Title</h1>
-    HTML
-    flow.driver.add_cookie("hello", "world")
-    flow.driver.get_cookie("hello").should eq "world"
+    it "can reset the session", tags: "headless_chrome" do
+      flow = visit_page_with <<-HTML
+        <h1>Title</h1>
+      HTML
+      flow.driver.add_cookie("hello", "world")
+      flow.driver.get_cookie("hello").should eq "world"
 
-    LuckyFlow.reset
+      LuckyFlow.reset
 
-    expect_raises Selenium::Error do
-      flow.driver.get_cookie("hello")
+      expect_raises Selenium::Error do
+        flow.driver.get_cookie("hello")
+      end
+    end
+
+    it "can accept and dismiss alerts", tags: "headless_chrome" do
+      flow = visit_page_with <<-HTML
+        <button onclick="createAlert()" flow-id="button" data-count="0">Click Me - 0</button>
+        <script>
+        function createAlert() {
+          alert("Are you sure?");
+          const button = document.querySelector("[flow-id='button']");
+          button.innerText = 'Click Me - ' + (++button.dataset.count);
+        }
+        </script>
+      HTML
+
+      flow.click("@button")
+      flow.accept_alert
+      flow.should have_element("@button", text: "Click Me - 1")
+      flow.click("@button")
+      flow.dismiss_alert
+      flow.should have_element("@button", text: "Click Me - 2")
     end
   end
 
-  it "can accept and dismiss alerts", tags: "headless_chrome" do
-    flow = visit_page_with <<-HTML
-      <button onclick="createAlert()" flow-id="button" data-count="0">Click Me - 0</button>
-      <script>
-      function createAlert() {
-        alert("Are you sure?");
-        const button = document.querySelector("[flow-id='button']");
-        button.innerText = 'Click Me - ' + (++button.dataset.count);
-      }
-      </script>
-    HTML
+  context "with headless firefox" do
+    it "can open screenshots", tags: "headless_firefox" do
+      flow = LuckyFlow.new
+      fake_process = FakeProcess
+      time = Time.utc
 
-    flow.click("@button")
-    flow.accept_alert
-    flow.should have_element("@button", text: "Click Me - 1")
-    flow.click("@button")
-    flow.dismiss_alert
-    flow.should have_element("@button", text: "Click Me - 2")
+      flow.open_screenshot(fake_process, time)
+
+      fake_process.shell.should be_true
+      fake_process.command.should eq "open ./tmp/screenshots/#{time.to_unix}.png"
+    end
+
+    it "can open fullsize screenshots", tags: "headless_firefox" do
+      flow = LuckyFlow.new
+      fake_process = FakeProcess
+      time = Time.utc
+
+      flow.open_screenshot(fake_process, time, fullsize: true)
+
+      fake_process.shell.should be_true
+      fake_process.command.should eq "open ./tmp/screenshots/#{time.to_unix}.png"
+    end
+
+    it "can reset the session", tags: "headless_firefox" do
+      flow = visit_page_with <<-HTML
+        <h1>Title</h1>
+      HTML
+      flow.driver.add_cookie("hello", "world")
+      flow.driver.get_cookie("hello").should eq "world"
+
+      LuckyFlow.reset
+
+      expect_raises Selenium::Error do
+        flow.driver.get_cookie("hello")
+      end
+    end
+
+    it "can accept and dismiss alerts", tags: "headless_firefox" do
+      flow = visit_page_with <<-HTML
+        <button onclick="createAlert()" flow-id="button" data-count="0">Click Me - 0</button>
+        <script>
+        function createAlert() {
+          alert("Are you sure?");
+          const button = document.querySelector("[flow-id='button']");
+          button.innerText = 'Click Me - ' + (++button.dataset.count);
+        }
+        </script>
+      HTML
+
+      flow.click("@button")
+      flow.accept_alert
+      flow.should have_element("@button", text: "Click Me - 1")
+      flow.click("@button")
+      flow.dismiss_alert
+      flow.should have_element("@button", text: "Click Me - 2")
+    end
   end
 
   it "can choose option in select input" do
@@ -340,7 +400,7 @@ describe LuckyFlow do
     end
   end
 
-  it "can hover over an element", tags: "headless_chrome" do
+  it "can hover over an element", tags: "headless_firefox" do
     flow = visit_page_with <<-HTML
       <style>
         #hidden {
diff --git a/src/lucky_flow/selenium/chrome/driver.cr b/src/lucky_flow/selenium/chrome/driver.cr
index ee52fc8..4f39a56 100644
--- a/src/lucky_flow/selenium/chrome/driver.cr
+++ b/src/lucky_flow/selenium/chrome/driver.cr
@@ -12,8 +12,8 @@ module LuckyFlow::Selenium
 
     private def driver_path
       LuckyFlow.settings.driver_path || Webdrivers::Chromedriver.install
-    rescue err
-      raise DriverInstallationError.new(err)
+    rescue e
+      raise DriverInstallationError.new(e)
     end
   end
 end
@@ -24,7 +24,13 @@ end
 
 LuckyFlow::Registry.register :headless_chrome do
   LuckyFlow::Selenium::Chrome::Driver.new do |config|
-    remote_debuggin_port = ENV["CHROME_REMOTE_DEBUGGING_PORT"]? || 9222
-    config.chrome_options.args = ["no-sandbox", "headless", "disable-gpu", "remote-debugging-port=#{remote_debuggin_port}"]
+    remote_debugging_port = ENV.fetch("CHROME_REMOTE_DEBUGGING_PORT", "9222")
+    config.chrome_options.args = [
+      "--no-sandbox",
+      "--headless",
+      "--disable-gpu",
+      "--disable-dev-shm-usage",
+      "--remote-debugging-port=#{remote_debugging_port}",
+    ]
   end
 end
diff --git a/src/lucky_flow/selenium/driver.cr b/src/lucky_flow/selenium/driver.cr
index 137eaa4..720395b 100644
--- a/src/lucky_flow/selenium/driver.cr
+++ b/src/lucky_flow/selenium/driver.cr
@@ -96,8 +96,14 @@ abstract class LuckyFlow::Selenium::Driver < LuckyFlow::Driver
     end
   end
 
-  private def find_elements(strategy : Symbol, query : String) : Array(LuckyFlow::Element)
-    session.find_elements(strategy, query)
-      .map { |el| LuckyFlow::Selenium::Element.new(self, query, el).as(LuckyFlow::Element) }
+  private def find_elements(
+    strategy : Symbol,
+    query : String
+  ) : Array(LuckyFlow::Element)
+    session.find_elements(strategy, query).map do |element|
+      LuckyFlow::Selenium::Element
+        .new(self, query, element)
+        .as(LuckyFlow::Element)
+    end
   end
 end
diff --git a/src/lucky_flow/selenium/firefox/driver.cr b/src/lucky_flow/selenium/firefox/driver.cr
new file mode 100644
index 0000000..791af22
--- /dev/null
+++ b/src/lucky_flow/selenium/firefox/driver.cr
@@ -0,0 +1,37 @@
+module LuckyFlow::Selenium
+  class Firefox::Driver < Driver
+    private getter driver : ::Selenium::Driver do
+      service = ::Selenium::Service.firefox(driver_path: driver_path)
+      ::Selenium::Driver.for(:firefox, service: service)
+    end
+
+    def initialize(&)
+      super ::Selenium::Firefox::Capabilities.new
+      yield @capabilities.as(::Selenium::Firefox::Capabilities)
+    end
+
+    private def driver_path
+      LuckyFlow.settings.driver_path || Webdrivers::Geckodriver.install
+    rescue e
+      raise DriverInstallationError.new(e)
+    end
+  end
+end
+
+LuckyFlow::Registry.register :firefox do
+  LuckyFlow::Selenium::Firefox::Driver.new { }
+end
+
+LuckyFlow::Registry.register :headless_firefox do
+  LuckyFlow::Selenium::Firefox::Driver.new do |config|
+    remote_debugging_port = ENV.fetch("FIREFOX_REMOTE_DEBUGGING_PORT", "9222")
+    config.firefox_options.args = [
+      "--no-sandbox",
+      "--headless",
+      "--disable-gpu",
+      "--disable-software-rasterizer",
+      "--disable-dev-shm-usage",
+      "--start-debugger-server=#{remote_debugging_port}",
+    ]
+  end
+end
diff --git a/src/lucky_flow/webless/driver.cr b/src/lucky_flow/webless/driver.cr
index d47e5fe..4fec7e7 100644
--- a/src/lucky_flow/webless/driver.cr
+++ b/src/lucky_flow/webless/driver.cr
@@ -46,11 +46,11 @@ class LuckyFlow::Webless::Driver < LuckyFlow::Driver
   end
 
   def find_css(query : String) : Array(LuckyFlow::Element)
-    @browser.find_css(query).map { |el| element(query, el) }
+    @browser.find_css(query).map { |elem| element(query, elem) }
   end
 
   def find_xpath(query : String) : Array(LuckyFlow::Element)
-    @browser.find_xpath(query).map { |el| element(query, el) }
+    @browser.find_xpath(query).map { |elem| element(query, elem) }
   end
 
   def current_url : String
diff --git a/src/lucky_flow/webless/element.cr b/src/lucky_flow/webless/element.cr
index b43f27d..01ddd12 100644
--- a/src/lucky_flow/webless/element.cr
+++ b/src/lucky_flow/webless/element.cr
@@ -77,7 +77,7 @@ class LuckyFlow::Webless::Element < LuckyFlow::Element
   end
 
   def remove_attribute(name : String, element : HTML5::Node = @inner_element)
-    element.attr.reject! { |at| at.key == name }
+    element.attr.reject! { |attr| attr.key == name }
   end
 
   def property(name : String) : String?
@@ -96,8 +96,9 @@ class LuckyFlow::Webless::Element < LuckyFlow::Element
     @inner_element.xpath_nodes(".//option[@selected]")
       .each { |node| remove_attribute("selected", node) }
 
-    @inner_element.xpath(".//option[@value='#{value}']")
-      .try { |el| find_or_create_attr("selected", el).val = "selected" }
+    @inner_element.xpath(".//option[@value='#{value}']").try do |elem|
+      find_or_create_attr("selected", elem).val = "selected"
+    end
   end
 
   def select_options(values : Array(String))
@@ -107,8 +108,9 @@ class LuckyFlow::Webless::Element < LuckyFlow::Element
       .each { |node| remove_attribute("selected", node) }
 
     values.each do |value|
-      @inner_element.xpath(".//option[@value='#{value}']")
-        .try { |el| find_or_create_attr("selected", el).val = "selected" }
+      @inner_element.xpath(".//option[@value='#{value}']").try do |elem|
+        find_or_create_attr("selected", elem).val = "selected"
+      end
     end
   end