Refaktoring kodu

Opakowanie w klasę