fixes to batches and drop / drill solving in mnist example