Mały refaktoring i dodanie kluczowych klas.