Code

#Updated graph generation