Add test for batch functionality
Remove unneeded comment line