Dirty working version. Switched to chrome webdriver for speed