Update hungarian2.py