diff --git a/java/src/main/java/org/softwareheritage/graph/maps/LabelMapBuilder.java b/java/src/main/java/org/softwareheritage/graph/maps/LabelMapBuilder.java
index 3f97a82..47905cd 100644
--- a/java/src/main/java/org/softwareheritage/graph/maps/LabelMapBuilder.java
+++ b/java/src/main/java/org/softwareheritage/graph/maps/LabelMapBuilder.java
@@ -1,425 +1,541 @@
 package org.softwareheritage.graph.maps;
 
 import com.martiansoftware.jsap.*;
 import it.unimi.dsi.big.webgraph.LazyLongIterator;
 import it.unimi.dsi.big.webgraph.labelling.ArcLabelledImmutableGraph;
 import it.unimi.dsi.big.webgraph.labelling.BitStreamArcLabelledImmutableGraph;
 import it.unimi.dsi.fastutil.BigArrays;
 import it.unimi.dsi.fastutil.Size64;
 import it.unimi.dsi.fastutil.bytes.ByteArrays;
 import it.unimi.dsi.fastutil.io.BinIO;
 import it.unimi.dsi.fastutil.io.FastBufferedInputStream;
 import it.unimi.dsi.fastutil.longs.LongBigArrays;
 import it.unimi.dsi.fastutil.longs.LongIterator;
 import it.unimi.dsi.fastutil.objects.Object2LongFunction;
 import it.unimi.dsi.io.OutputBitStream;
 import it.unimi.dsi.logging.ProgressLogger;
 import it.unimi.dsi.big.webgraph.BVGraph;
 import it.unimi.dsi.big.webgraph.ImmutableGraph;
 import it.unimi.dsi.big.webgraph.NodeIterator;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.softwareheritage.graph.labels.DirEntry;
 import org.softwareheritage.graph.labels.SwhLabel;
 
 import java.io.*;
 import java.lang.reflect.Array;
+import java.math.BigInteger;
+import java.nio.ByteBuffer;
+import java.nio.charset.Charset;
 import java.nio.charset.StandardCharsets;
 import java.util.*;
 import java.util.concurrent.TimeUnit;
 
 public class LabelMapBuilder {
     final static String SORT_BUFFER_SIZE = "40%";
     final static Logger logger = LoggerFactory.getLogger(LabelMapBuilder.class);
 
     String graphPath;
     String debugPath;
     String tmpDir;
     ImmutableGraph graph;
 
     Object2LongFunction<byte[]> swhIdMph;
     long[][] orderMap;
     Object2LongFunction<byte[]> filenameMph;
     long numFilenames;
     int totalLabelWidth;
 
     public LabelMapBuilder(String graphPath, String debugPath, String tmpDir) {
         this.graphPath = graphPath;
         this.debugPath = debugPath;
         this.tmpDir = tmpDir;
     }
 
     private static JSAPResult parse_args(String[] args) {
         JSAPResult config = null;
         try {
             SimpleJSAP jsap = new SimpleJSAP(LabelMapBuilder.class.getName(), "",
                     new Parameter[]{
                             new FlaggedOption("graphPath", JSAP.STRING_PARSER, JSAP.NO_DEFAULT, JSAP.REQUIRED, 'g',
                                     "graph", "Basename of the compressed graph"),
                             new FlaggedOption("debugPath", JSAP.STRING_PARSER, JSAP.NO_DEFAULT, JSAP.NOT_REQUIRED, 'd',
                                     "debug-path", "Store the intermediate representation here for debug"),
 
                             new FlaggedOption("tmpDir", JSAP.STRING_PARSER, "tmp", JSAP.NOT_REQUIRED, 't', "tmp",
                                     "Temporary directory path"),});
 
             config = jsap.parse(args);
             if (jsap.messagePrinted()) {
                 System.exit(1);
             }
         } catch (JSAPException e) {
             e.printStackTrace();
         }
         return config;
     }
 
     public static void main(String[] args) throws IOException, InterruptedException {
         JSAPResult config = parse_args(args);
         String graphPath = config.getString("graphPath");
         String tmpDir = config.getString("tmpDir");
         String debugPath = config.getString("debugPath");
 
         LabelMapBuilder builder = new LabelMapBuilder(graphPath, debugPath, tmpDir);
 
         logger.info("Loading graph and MPH functions...");
         builder.loadGraph();
 
-        builder.computeLabelMapSort();
+        // builder.computeLabelMapSort();
+        builder.computeLabelMapBsort();
     }
 
     @SuppressWarnings("unchecked") // Suppress warning for Object2LongFunction cast
     static Object2LongFunction<byte[]> loadMPH(String mphBasename) throws IOException {
         Object2LongFunction<byte[]> mphMap = null;
         try {
             mphMap = (Object2LongFunction<byte[]>) BinIO.loadObject(mphBasename + ".mph");
         } catch (ClassNotFoundException e) {
             logger.error("unknown class object in .mph file: " + e);
             System.exit(2);
         }
         return mphMap;
     }
 
     static long getMPHSize(Object2LongFunction<byte[]> mph) {
         return (mph instanceof Size64) ? ((Size64) mph).size64() : mph.size();
     }
 
     void computeLabelMapSort() throws IOException {
         // Pass the intermediate representation to sort(1) so that we see the labels in the order they will
         // appear in the label file.
         ProcessBuilder processBuilder = new ProcessBuilder();
         processBuilder.command("sort", "-k1,1n", "-k2,2n", // Numerical sort
                 "--numeric-sort", "--buffer-size", SORT_BUFFER_SIZE, "--temporary-directory", tmpDir);
         Process sort = processBuilder.start();
         BufferedOutputStream sort_stdin = new BufferedOutputStream(sort.getOutputStream());
-        BufferedInputStream sort_stdout = new BufferedInputStream(sort.getInputStream());
+        // BufferedInputStream sort_stdout = new BufferedInputStream(sort.getInputStream());
+        FastBufferedInputStream sort_stdout = new FastBufferedInputStream(sort.getInputStream());
 
         final FastBufferedInputStream fbis = new FastBufferedInputStream(System.in);
         hashLabelStream(fbis, new EdgeLabelLineWriter() {
             @Override
             public void writeLine(long src, long dst, long filenameId, int permission) throws IOException {
                 sort_stdin.write((src + "\t" + dst + "\t" + filenameId + "\t" + permission + "\n")
                         .getBytes(StandardCharsets.US_ASCII));
             }
         });
         sort_stdin.close();
 
         EdgeLabelLineIterator mapLines = new TextualEdgeLabelLineIterator(sort_stdout);
         writeLabels(mapLines);
-
         logger.info("Done");
     }
 
     void computeLabelMapBsort() throws IOException, InterruptedException {
         // Pass the intermediate representation to bsort(1) so that we see the labels in the order they will
         // appear in the label file.
+
         String tmpFile = tmpDir + "/labelsToSort.bin";
         final FastBufferedInputStream fbis = new FastBufferedInputStream(System.in);
         final DataOutputStream out = new DataOutputStream(new BufferedOutputStream(new FileOutputStream(tmpFile)));
+
+        // Number of bytes to represent a node.
+        final int nodeBytes = (Long.SIZE - Long.numberOfLeadingZeros(graph.numNodes())) / 8 + 1;
+        ByteBuffer buffer = ByteBuffer.allocate(Long.BYTES);
+        logger.info("Writing labels to a packed binary files (node bytes: {})", nodeBytes);
+
         hashLabelStream(fbis, new EdgeLabelLineWriter() {
             @Override
             public void writeLine(long src, long dst, long filenameId, int permission) throws IOException {
-                out.writeLong(src);
-                out.writeLong(dst);
+                buffer.putLong(0, src);
+                out.write(buffer.array(), Long.BYTES - nodeBytes, nodeBytes);
+                buffer.putLong(0, dst);
+                out.write(buffer.array(), Long.BYTES - nodeBytes, nodeBytes);
                 out.writeLong(filenameId);
                 out.writeInt(permission);
             }
         });
 
         ProcessBuilder processBuilder = new ProcessBuilder();
         processBuilder.command(
             "/home/seirl/bsort/src/bsort",
             "-v",
-            "-r", String.valueOf(8 + 8 + 8 + 4),
-            "-k", String.valueOf(8 + 8),
+            "-r", String.valueOf(nodeBytes * 2 + Long.BYTES + Integer.BYTES),
+            "-k", String.valueOf(nodeBytes * 2),
             tmpFile
         );
         Process sort = processBuilder.start();
         sort.waitFor();
 
-        final DataInputStream sortedLabels = new DataInputStream(new DataInputStream(new FileInputStream(tmpFile)));
-        BinaryEdgeLabelLineIterator mapLines = new BinaryEdgeLabelLineIterator(sortedLabels);
+        final DataInputStream sortedLabels = new DataInputStream(new BufferedInputStream(new FileInputStream(tmpFile)));
+        BinaryEdgeLabelLineIterator mapLines = new BinaryEdgeLabelLineIterator(sortedLabels, nodeBytes);
         writeLabels(mapLines);
 
         logger.info("Done");
     }
 
     void loadGraph() throws IOException {
         graph = BVGraph.loadMapped(graphPath);
 
         swhIdMph = loadMPH(graphPath);
 
         orderMap = LongBigArrays.newBigArray(getMPHSize(swhIdMph));
         BinIO.loadLongs(graphPath + ".order", orderMap);
 
         filenameMph = loadMPH(graphPath + "-labels");
         numFilenames = getMPHSize(filenameMph);
         totalLabelWidth = DirEntry.labelWidth(numFilenames);
     }
 
     void hashLabelStream(FastBufferedInputStream input, EdgeLabelLineWriter writer) throws IOException {
         // Compute intermediate representation and write it on :
         // "<src node id> <dst node id> <label ids>\n"
         ProgressLogger plInter = new ProgressLogger(logger, 10, TimeUnit.SECONDS);
         plInter.itemsName = "edges";
         plInter.expectedUpdates = graph.numArcs();
         plInter.start("Hashing the label stream");
 
         var charset = StandardCharsets.US_ASCII;
         byte[] array = new byte[1024];
         for (long line = 0;; line++) {
             int start = 0, len;
             while ((len = input.readLine(array, start, array.length - start,
                     FastBufferedInputStream.ALL_TERMINATORS)) == array.length - start) {
                 start += len;
                 array = ByteArrays.grow(array, array.length + 1);
             }
             if (len == -1)
                 break; // EOF
             final int lineLength = start + len;
 
             // Skip whitespace at the start of the line.
             int offset = 0;
             while (offset < lineLength && array[offset] >= 0 && array[offset] <= ' ')
                 offset++;
             if (offset == lineLength) {
                 continue;
             }
             if (array[0] == '#')
                 continue;
 
             // Scan source id.
             start = offset;
             while (offset < lineLength && (array[offset] < 0 || array[offset] > ' '))
                 offset++;
             final byte[] ss = Arrays.copyOfRange(array, start, offset);
 
             // Skip whitespace between identifiers.
             while (offset < lineLength && array[offset] >= 0 && array[offset] <= ' ')
                 offset++;
             if (offset == lineLength) {
                 logger.error("Error at line " + line + ": no target");
                 continue;
             }
 
             // Scan target ID
             start = offset;
             while (offset < lineLength && (array[offset] < 0 || array[offset] > ' '))
                 offset++;
             final byte[] ts = Arrays.copyOfRange(array, start, offset);
 
             // Skip whitespace between identifiers.
             while (offset < lineLength && array[offset] >= 0 && array[offset] <= ' ')
                 offset++;
             if (offset == lineLength)
                 continue; // No label, skip
 
             // Scan label
             start = offset;
             while (offset < lineLength && (array[offset] < 0 || array[offset] > ' '))
                 offset++;
             final byte[] ls = Arrays.copyOfRange(array, start, offset);
 
             // Skip whitespace between identifiers.
             while (offset < lineLength && array[offset] >= 0 && array[offset] <= ' ')
                 offset++;
             // Scan permission
             int permission = 0;
             if (offset < lineLength) {
                 start = offset;
                 while (offset < lineLength && (array[offset] < 0 || array[offset] > ' '))
                     offset++;
                 permission = Integer.parseInt(new String(array, start, offset - start, charset));
             }
 
             // System.err.format("DEBUG: read %s %s %s %d\n", ss, ts, ls, permission);
 
             long s = swhIdMph.getLong(ss);
             long t = swhIdMph.getLong(ts);
             long filenameId = filenameMph.getLong(ls);
 
             long srcNode = BigArrays.get(orderMap, s);
             long dstNode = BigArrays.get(orderMap, t);
 
             writer.writeLine(srcNode, dstNode, filenameId, permission);
             plInter.lightUpdate();
         }
         plInter.done();
     }
 
     void writeLabels(EdgeLabelLineIterator mapLines) throws IOException {
         // Get the sorted output and write the labels and label offsets
         ProgressLogger plLabels = new ProgressLogger(logger, 10, TimeUnit.SECONDS);
         plLabels.itemsName = "edges";
         plLabels.expectedUpdates = graph.numArcs();
-        plLabels.start("Hashing the label stream");
+        plLabels.start("Writing the labels to the label file.");
 
         FileWriter debugFile = null;
         if (debugPath != null) {
             debugFile = new FileWriter(debugPath);
         }
 
         OutputBitStream labels = new OutputBitStream(
                 new File(graphPath + "-labelled" + BitStreamArcLabelledImmutableGraph.LABELS_EXTENSION));
         OutputBitStream offsets = new OutputBitStream(
                 new File(graphPath + "-labelled" + BitStreamArcLabelledImmutableGraph.LABEL_OFFSETS_EXTENSION));
         offsets.writeGamma(0);
 
         EdgeLabelLine line = new EdgeLabelLine(-1, -1, -1, -1);
 
         NodeIterator it = graph.nodeIterator();
         boolean started = false;
 
         ArrayList<DirEntry> labelBuffer = new ArrayList<>(128);
         while (it.hasNext()) {
             long srcNode = it.nextLong();
-            if (!started) {
-                plLabels.start("Writing label map to file...");
-                started = true;
-            }
 
             int bits = 0;
             LazyLongIterator s = it.successors();
             long dstNode;
             while ((dstNode = s.nextLong()) >= 0) {
-                while (line.srcNode <= srcNode && line.dstNode <= dstNode) {
+                while (line != null && line.srcNode <= srcNode && line.dstNode <= dstNode) {
                     if (line.srcNode == srcNode && line.dstNode == dstNode) {
                         labelBuffer.add(new DirEntry(line.filenameId, line.permission));
 
                         if (debugFile != null) {
                             debugFile.write(line.srcNode + " " + line.dstNode + " " + line.filenameId + " "
                                     + line.permission + "\n");
                         }
                     }
 
                     if (!mapLines.hasNext())
                         break;
 
                     line = mapLines.next();
-                    if (line == null)
-                        break;
+                    if (!started) {
+                        plLabels.start("Writing label map to file...");
+                        started = true;
+                    }
                 }
 
                 SwhLabel l = new SwhLabel("edgelabel", totalLabelWidth, labelBuffer.toArray(new DirEntry[0]));
                 labelBuffer.clear();
                 bits += l.toBitStream(labels, -1);
+                plLabels.lightUpdate();
             }
             offsets.writeGamma(bits);
-            plLabels.lightUpdate();
         }
 
         labels.close();
         offsets.close();
         plLabels.done();
         if (debugFile != null) {
             debugFile.close();
         }
 
         PrintWriter pw = new PrintWriter(new FileWriter((new File(graphPath)).getName() + "-labelled.properties"));
         pw.println(ImmutableGraph.GRAPHCLASS_PROPERTY_KEY + " = " + BitStreamArcLabelledImmutableGraph.class.getName());
         pw.println(BitStreamArcLabelledImmutableGraph.LABELSPEC_PROPERTY_KEY + " = " + SwhLabel.class.getName()
                 + "(DirEntry," + totalLabelWidth + ")");
         pw.println(ArcLabelledImmutableGraph.UNDERLYINGGRAPH_PROPERTY_KEY + " = " + graphPath);
         pw.close();
     }
 
     public abstract static class EdgeLabelLineWriter {
         public abstract void writeLine(long src, long dst, long filenameId, int permission) throws IOException;
     }
 
     public static class EdgeLabelLine {
         public long srcNode;
         public long dstNode;
         public long filenameId;
         public int permission;
 
         public EdgeLabelLine(long labelSrcNode, long labelDstNode, long labelFilenameId, int labelPermission) {
             this.srcNode = labelSrcNode;
             this.dstNode = labelDstNode;
             this.filenameId = labelFilenameId;
             this.permission = labelPermission;
         }
     }
 
     public abstract static class EdgeLabelLineIterator implements Iterator<EdgeLabelLine> {
         @Override
         public abstract boolean hasNext();
         @Override
         public abstract EdgeLabelLine next();
     }
 
-    public static class TextualEdgeLabelLineIterator extends EdgeLabelLineIterator {
+    public static class ScannerEdgeLabelLineIterator extends EdgeLabelLineIterator {
         private final Scanner scanner;
 
-        public TextualEdgeLabelLineIterator(InputStream stream) {
-            scanner = new Scanner(stream, StandardCharsets.US_ASCII);
+        public ScannerEdgeLabelLineIterator(InputStream input) {
+            this.scanner = new Scanner(input);
         }
 
         @Override
         public boolean hasNext() {
-            return scanner.hasNext();
+            return this.scanner.hasNext();
         }
 
         @Override
         public EdgeLabelLine next() {
             String line = scanner.nextLine();
             String[] parts = line.split("\\t");
             return new EdgeLabelLine(
-                Long.parseLong(parts[0]),
-                Long.parseLong(parts[1]),
-                Long.parseLong(parts[2]),
-                Integer.parseInt(parts[3])
+                    Long.parseLong(parts[0]),
+                    Long.parseLong(parts[1]),
+                    Long.parseLong(parts[2]),
+                    Integer.parseInt(parts[3])
             );
+
+            /*
+            String line = scanner.nextLine();
+            long src = scanner.nextLong();
+            long dst = scanner.nextLong();
+            long label = scanner.nextLong();
+            int permission = scanner.nextInt();
+            return new EdgeLabelLine(src, dst, label, permission);
+             */
+        }
+    }
+
+    public static class TextualEdgeLabelLineIterator extends EdgeLabelLineIterator {
+        private final FastBufferedInputStream input;
+        private final Charset charset;
+        private byte[] array;
+        boolean finished;
+
+        public TextualEdgeLabelLineIterator(FastBufferedInputStream input) {
+            this.input = input;
+            this.finished = false;
+            this.charset = StandardCharsets.US_ASCII;
+            this.array = new byte[1024];
+        }
+
+        @Override
+        public boolean hasNext() {
+            return !this.finished;
+        }
+
+        @Override
+        public EdgeLabelLine next() {
+            int start = 0, len;
+            try {
+                while ((len = input.readLine(array, start, array.length - start,
+                        FastBufferedInputStream.ALL_TERMINATORS)) == array.length - start) {
+                    start += len;
+                    array = ByteArrays.grow(array, array.length + 1);
+                }
+            } catch (IOException e) {
+                e.printStackTrace();
+                return null;
+            }
+            if (len == -1) {
+                finished = true;
+                return null;
+            }
+            final int lineLength = start + len;
+
+            int offset = 0;
+
+            // Scan source id.
+            start = offset;
+            while (offset < lineLength && (array[offset] < 0 || array[offset] > ' '))
+                offset++;
+            long src = Long.parseLong(new String(array, start, offset - start, charset));
+
+            // Skip whitespace between identifiers.
+            while (offset < lineLength && array[offset] >= 0 && array[offset] <= ' ')
+                offset++;
+
+            // Scan target ID
+            start = offset;
+            while (offset < lineLength && (array[offset] < 0 || array[offset] > ' '))
+                offset++;
+            long dst = Long.parseLong(new String(array, start, offset - start, charset));
+
+            // Skip whitespace between identifiers.
+            while (offset < lineLength && array[offset] >= 0 && array[offset] <= ' ')
+                offset++;
+
+            // Scan label
+            start = offset;
+            while (offset < lineLength && (array[offset] < 0 || array[offset] > ' '))
+                offset++;
+            long label = Long.parseLong(new String(array, start, offset - start, charset));
+
+            // Skip whitespace between identifiers.
+            while (offset < lineLength && array[offset] >= 0 && array[offset] <= ' ')
+                offset++;
+
+            // Scan permission
+            int permission = 0;
+            if (offset < lineLength) {
+                start = offset;
+                while (offset < lineLength && (array[offset] < 0 || array[offset] > ' '))
+                    offset++;
+                permission = Integer.parseInt(new String(array, start, offset - start, charset));
+            }
+
+            return new EdgeLabelLine(src, dst, label, permission);
         }
     }
 
     public static class BinaryEdgeLabelLineIterator extends EdgeLabelLineIterator {
+        private final int nodeBytes;
         DataInputStream stream;
         boolean finished;
+        ByteBuffer buffer;
 
-        public BinaryEdgeLabelLineIterator(DataInputStream stream) {
+        public BinaryEdgeLabelLineIterator(DataInputStream stream, int nodeBytes) {
             this.stream = stream;
+            this.nodeBytes = nodeBytes;
+            this.buffer = ByteBuffer.allocate(Long.BYTES);
             this.finished = false;
         }
 
         @Override
         public boolean hasNext() {
-            return finished;
+            return !finished;
         }
 
         @Override
         public EdgeLabelLine next() {
             try {
-                long src = stream.readLong();
-                long dst = stream.readLong();
+                // long src = stream.readLong();
+                // long dst = stream.readLong();
+                stream.readFully(this.buffer.array(), Long.BYTES - nodeBytes, nodeBytes);
+                this.buffer.position(0);
+                long src = this.buffer.getLong();
+                this.buffer.clear();
+
+                stream.readFully(this.buffer.array(), Long.BYTES - nodeBytes, nodeBytes);
+                this.buffer.position(0);
+                long dst = this.buffer.getLong();
+                this.buffer.clear();
+
                 long label = stream.readLong();
                 int perm = stream.readInt();
                 return new EdgeLabelLine(src, dst, label, perm);
             } catch (EOFException e) {
                 finished = true;
                 return null;
             } catch (IOException e) {
                 e.printStackTrace();
                 return null;
             }
         }
     }
 }