Refactor location of training files