Wrote in description of code in hw2.py