Update file naming scheme
Fix getting labels and docs on smaller datasets