fix a small bug to lab3.py