diff --git a/lib/APPReinforcementLearning/src/ReadyState.h b/lib/APPReinforcementLearning/src/ReadyState.h
index 93949f3..bb827e5 100644
--- a/lib/APPReinforcementLearning/src/ReadyState.h
+++ b/lib/APPReinforcementLearning/src/ReadyState.h
@@ -110,7 +110,7 @@ class ReadyState : public IState
     /**
      *  Set the selected mode.
      *
-     * @return It returns 1 if DrivingMode is selected or 2 if TrainingMode is selected, otherwise 0.
+     * @return It returns the selected Mode.
      */
     Mode getSelectedMode();
 
diff --git a/lib/APPReinforcementLearning/src/TrainingState.h b/lib/APPReinforcementLearning/src/TrainingState.h
index 737d7a8..a5ad381 100644
--- a/lib/APPReinforcementLearning/src/TrainingState.h
+++ b/lib/APPReinforcementLearning/src/TrainingState.h
@@ -131,7 +131,3 @@ class TrainingState : public IState
 
 #endif /* TRAINING_STATE_H */
 /** @} */
-
-/******************************************************************************
- * Local Functions
- *****************************************************************************/