Trochę refaktoryzacji ale nie wiele