added test datasets, updated code a little, updated setup file to create test dataset