old-zsterminator/app/controllers/application_controller.rb

class ApplicationController < ActionController::Base

  private
  def set_company
    company_id = session.fetch(:company_id, Company.first&.id)
    session[:company_id] = company_id
    @company = Company.find(session[:company_id])
  end
end