I think I get it now. --nmax always rounds down to the nearest cube (nxnxn) where n is an integer >= 1. So the results from Fig. 4 must be from a 216-atom (3x3x3) supercell not a 4x4x4 primitive supercell. Please correct me if I am wrong.
Using a 216-atom supercell for GaAs, I get similar results for Fig. 4a and Fig. 4b. The results I got compared to Fig. 4b appear to be a bit different, for example, there is only Vq/b not Vq/b for both As and Ga. If someone can tell me why that is that would be great. Also, it matches in charge correction data in SI2 fairly well.
I've attached some of my plots below. The formation energies in Fig. 5 however appear to match my results more for the 64-atom (2x2x2) supercell rather than the 216-atom (3x3x3) supercell.