Add training and test data

Signed-off-by: Nicholas Kamper <kampernj@rose-hulman.edu>