Refactor label, document, and feature retrieval
Rename function
Add support for smaller data set creation