Chagne the WatchTrain function slightly.