new approach
create subset of big dataset for testing purposes