fixed a bug in assignment11.py