updated to hw6 pt 1