added trainTracker todo