add grid index finder to normalization.py