finished hw5, I think