Refactor "TopOneTrain"