updated classdiagram