Working on homework 2