Test and Train partitions added.
Added test and train sets.