 				
 				var defaultSize:CGFloat = 70
 				
-				if self.model.data.visSearchDifficulty == 1 { // Hard mode
+                if self.model.data.visSearchDifficulty == Mode.Hard.rawValue {
 					defaultSize = 54
 				}
 				

					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 {

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