Update position_assign.py