inny pomysł, uproszczenie kodu i optymalizacja