Added MockDriver class