Add auto-registration of message handlers