Modify homework D4