fixed train val test files and python setup script