Updated TrainData.php and fixed remote branch