Complete 'lab5.py'
Remove unnecessary variables