Updated incomplete documentation for the on_message callback.