Change data subsetting

Better way of removing NA rows in the data.