clean up training
reorganize