Fixing up 1st hw