Added a smaller testing set