Renamed GameSelectionState to TrainingState
More tweaks