Create plot_roots

Function to plot correct roots
Edit plotting