Add support for `message` function, fixes #139