Update lab2.py

Created by chaveza (alejandro chavez)