Add some docs and assert