Added more data sets for testing