Setup Driver class