fixed errors with train output