Unit test generator working