hw1 updated though I know this is wrong