Update nb_step2_mapper.py