Fix bug and draw trained line on "Train until all N correct"