Update snapshotmodule to add batch functions, clean up the code

Update snapshotmodule to add batch functions, clean up the code #169