Added individual files for training and testing