Added context to Wit#message

This allows you to make use of the state feature