First draft hw6, still needs work