Some code updates, now randomly generating prediction graph