Add graph output option, need to update to dot format