Adds test driver class