Committing the changes I did yesterday

Just committing the changes I did yesterday for testing to see how I
could add data splitting.