Refactoring message module