Added some more tests for the code generator