Modified steps to allow WebDriver to be passed