More editing on 'lab2.py'