Added selenium import class