Add small test data set to test/data and add tests