Class diagram

Poprawiony class diagram