updated for hw6