classendiagram verbeterd en de code erachter