Ulepszenie istniejących funkcjonalności i trochę refaktoringu