Refaktorert litt. Boy scout style.