Easy graph generation and obtain statistics

to compare with current torflow results.