Finish homework 2