added Sk learn comparison for plotting