Remove selenium requirement, add mock dependency