cleaned version ready for running tests on random data to understand training and test error