Kolejne drobne ulepszenia i refactor.