get rid of test and train dirs