Added graph creator and visualization