Updated tacc train script.