Simplify the find_in_batches implementation a bit and put in some specs