using small dataset (for memory check)