TRAIN THE ALGORITHM