Update to homwework4 part1