Add test driver. Ugly though.