Added a test code generator