finished up trainig data reader
minor tweak