Update lab2.py

This should be right. The few behind this one have a couple typos