Refactoring: Added missing method