Add Capybara support