Add back batch loading after previous fix
Merge pull request #4 from dustin1114/master

Add back batch loading after previous fix