Added findCorners to train