Jak poprawić kod żeby był bardziej schludny ?