 				// Play the success sound
 				successSound.play()
 				
-				// Hard mode
 				var gameStage = currentView
-				if model.data.visSearchDifficulty == 1 { // Hard mode.
+                if model.data.visSearchDifficulty == Mode.Hard.rawValue {
 					gameStage -= VisualSearchHardModeView.TrainingOne.rawValue
 				}
 				

					self.isTraining = false
				}
				
				if self.model.data.visSearchDifficulty == 1 {  // Hard mode.
					if self.currentView != VisualSearchHardModeView.TrainingOne.rawValue
                        && self.currentView != VisualSearchHardModeView.TrainingTwo.rawValue
                        && self.currentView != VisualSearchHardModeView.TrainingThree.rawValue {

		collectionView.delegate = self
		collectionView.dataSource = self
		
		if model.data.visSearchDifficulty == Mode.Hard.rawValue {
			currentView = VisualSearchHardModeView.TrainingOne.rawValue
			numberOfTargets = [1, 1, 2, 9, 9, 9, 9]
			gameSpeed = model.data.visSearchSpeedHard.doubleValue
