Finish working on Junit test generation