Add option to take top <n> voxels in make_masks.py

@toddt how does this look