minor changes to lab8.py