Refaktoryzjacja kodu i poprawienie funkcjionalnosci