slight changes to lab5.py