fix some bugs in extenddisplay.py