Added smaller test dataset