Code refactor. Changed 2 spaces to 4