Batch predictions fixes, tests and docs