Add ability to only load data to limited states

Closes #44