Moved graph generation to more proper location