Some changes to the generated graph. Generating multiple files now