add missing assertion
add annotations