Improve message generator