Use whole dataset. Refactor code for choosing dataset sizes.