Added test/train data to repo