finished homework 2

uh not sure if this works but I tried ok