Importing Message as new in __init__.py. Also updated doc string.