Refactoring

Zlepseni kodu