Fix Little bug and add TrainingChart