script for making a subset of the data for testing