Poprawiony diagram klas