Poprawiony diagram klas.