Update disp2.py