Handle `Fe` vocab size correctly

[alam-shahul]

Right now, the Fe vocab size is automatically set as the number of genes + 2, which is correct for the Fg vocab size but totally incorrect for the Fes. This should be fixed.