More work on editing 'lab2.py'