Add test generator