added train test model directories