Fixed neural nets so that they no longer train and test the same data