Changed input to TChain for running on batch