Added small test set