Added plot to show the epoch training errors