Create function plot_learning_curve