class Selenium::WebDriver::ActionBuilder

def device(name: nil, type: nil)
  input = @devices.find { |device| (device.name == name.to_s || name.nil?) && (device.type == type || type.nil?) }
  raise(ArgumentError, "Can not find device: #{name}") if name && input.nil?
  input
end