Add messages to assert
Update docs