Add comment. Refactoring code.