initial commit
added code to subset dataset