Plotting functionality added