Only remaining is text output on plot. Currently training extensively