Refactoring: Removed unnecessary Unit return type