updated files to add logic for - Graph generation