code aangepast n.a.v. klassendiagram