Got some code to finish this hw.  hw4 done.