fixed lab3b.py