added support for capybara