updated driver class, deleted a test class