Testing smaller dataset