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