Fix plot_graph feature

[VedangW]

Fix graph plotting in plotly

[VedangW]

Graph feature fixed, need to add the plotting of the hyperplane.