added some tests to Generate.java to modify the output