jeden drobny błąd + mały code cleanup