adding the test and train directories to main directory