adding code to create a test subset