script works for all, adding file to test functionality for reading multiple files for training and testing