added training and test file