[hw_4] part 4 update